/*****************************************************************************
 *
 * This file is part of Mapnik (c++ mapping toolkit)
 *
 * Copyright (C) 2024 Artem Pavlenko
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 *****************************************************************************/

/** \brief Raster Colouriser
 *
 * This class allows GDAL raster bands to be colourised. It only works with single
 * band GDAL rasters, not greyscale, alpha, or rgb  (due to the GDAL featureset loading
 * single channel GDAL rasters as FLOAT32, and the others as BYTE, and not having a method
 * of figuring out which).
 *
 * Every input value is translated to an output value. The output value is determined by
 * what 'stop' the input value is in. Each stop covers the range of input values from its
 * 'value' parameter, up to the 'value' parameter of the next stop.
 *
 */

#ifndef MAPNIK_RASTER_COLORIZER_HPP
#define MAPNIK_RASTER_COLORIZER_HPP

// mapnik
#include <mapnik/config.hpp>
#include <mapnik/color.hpp>
#include <mapnik/enumeration.hpp>
#include <mapnik/image.hpp>

#include <memory>
#include <vector>
#include <optional>

namespace mapnik {

class feature_impl;
class raster;

//! \brief Enumerates the modes of interpolation
enum class colorizer_mode_enum : std::uint8_t {
    COLORIZER_INHERIT = 0,     //!< The stop inherits the mode from the colorizer
    COLORIZER_LINEAR = 1,      //!< Linear interpolation between colors, each channel separately
    COLORIZER_DISCRETE = 2,    //!< Single color for stop
    COLORIZER_EXACT = 3,       //!< Only the exact value specified for the stop gets translated, others use the default
    COLORIZER_LINEAR_RGBA = 4, //!< Linear interpolation between colors, all channels combined as RGBA value
    COLORIZER_LINEAR_BGRA = 5, //!< Linear interpolation between colors, all channels combined as BGRA value
    colorizer_mode_enum_MAX
};

DEFINE_ENUM(colorizer_mode, colorizer_mode_enum);

//! \brief Structure to represent a stop position.
class MAPNIK_DECL colorizer_stop
{
  public:

    //! \brief Constructor
    //!
    //! \param[in] value The stop value
    //! \param[in] mode The stop mode
    //! \param[in] _color The stop color
    colorizer_stop(float value = 0,
                   colorizer_mode mode = colorizer_mode_enum::COLORIZER_INHERIT,
                   color const& _color = color(0, 0, 0, 0),
                   std::string const& label = "");

    //! \brief Copy constructor
    colorizer_stop(colorizer_stop const& stop);

    //! \brief Destructor
    ~colorizer_stop();

    //! \brief Set the stop value
    //! \param[in] value The stop value
    inline void set_value(float v) { value_ = v; }

    //! \brief Get the stop value
    //! \return The stop value
    inline float get_value() const { return value_; }

    //! \brief Set the stop mode
    //! \param[in] mode The stop mode
    inline void set_mode(colorizer_mode mode) { mode_ = mode; }
    inline void set_mode_enum(colorizer_mode_enum mode) { set_mode(mode); }

    //! \brief Get the stop mode
    //! \return The stop mode
    inline colorizer_mode get_mode() const { return mode_; }
    inline colorizer_mode_enum get_mode_enum() const { return get_mode(); }

    //! \brief set the stop color
    //! \param[in] _color The stop color
    inline void set_color(color const& _color) { color_ = _color; }

    //! \brief get the stop color
    //! \return The stop color
    inline color const& get_color() const { return color_; }

    //! \brief set the stop label
    //! \param[in] label The stop label
    inline void set_label(std::string const& label) { label_ = label; }

    //! \brief get the stop label
    //! \return The stop label
    inline std::string const& get_label() const { return label_; }

    //! \brief Equality operator
    //! \return True if equal, false otherwise
    bool operator==(colorizer_stop const& other) const;

    //! \brief Print the stop to a string
    //! \return A string representing this stop.
    std::string to_string() const;

  private:
    float value_;         //!< The stop value
    colorizer_mode mode_; //!< The stop mode
    color color_;         //!< The stop color
    std::string label_;   //!< The stop label for use in legends
};

using colorizer_stops = std::vector<colorizer_stop>;

//! \brief Class representing the raster colorizer
class MAPNIK_DECL raster_colorizer
{
  public:
    //! \brief Constructor
    raster_colorizer(colorizer_mode mode = colorizer_mode_enum::COLORIZER_LINEAR,
                     color const& _color = color(0, 0, 0, 0));

    //! \brief Destructor
    ~raster_colorizer();

    //! \brief Set the default mode
    //!
    //! This can not be set as INHERIT, if you do, LINEAR will be used instead.
    //! \param[in] mode The default mode

    void set_default_mode(colorizer_mode mode)
    {
        default_mode_ =
          (mode == colorizer_mode_enum::COLORIZER_INHERIT) ? colorizer_mode_enum::COLORIZER_LINEAR : mode.value_;
    }

    void set_default_mode_enum(colorizer_mode_enum mode) { set_default_mode(mode); }

    //! \brief Get the default mode
    //! \return The default mode
    colorizer_mode get_default_mode() const { return default_mode_; }
    colorizer_mode_enum get_default_mode_enum() const { return get_default_mode(); }

    //! \brief Set the default color
    //! \param[in] color The default color
    void set_default_color(color const& color) { default_color_ = color; }

    //! \brief Get the default color
    //! \return The default color
    color const& get_default_color() const { return default_color_; }

    //! \brief Add a stop
    //!
    //! \param[in] stop The stop to add
    //! \return True if added, false if error
    bool add_stop(colorizer_stop const& stop);

    //! \brief Set the list of stops
    //! \param[in] stops The list of stops
    void set_stops(colorizer_stops const& stops) { stops_ = stops; }

    //! \brief Get the list of stops
    //! \return The list of stops
    colorizer_stops const& get_stops() const { return stops_; }

    template<typename T>
    void colorize(image_rgba8& out, T const& in, std::optional<double> const& nodata, feature_impl const& f) const;

    //! \brief Perform the translation of input to output
    //!
    //! \param[in] value Input value
    //! \return color associated with the value
    unsigned get_color(float v) const;

    //! \brief Set the epsilon value for exact mode
    //! \param[in] e The epsilon value
    inline void set_epsilon(const float e)
    {
        if (e > 0)
            epsilon_ = e;
    }

    //! \brief Get the epsilon value for exact mode
    //! \return The epsilon value
    inline float get_epsilon() const { return epsilon_; }

  private:
    colorizer_stops stops_; //!< The vector of stops

    colorizer_mode default_mode_; //!< The default mode inherited by stops
    color default_color_;         //!< The default color
    float epsilon_;               //!< The epsilon value for exact mode
};

using raster_colorizer_ptr = std::shared_ptr<raster_colorizer>;

} // namespace mapnik

#endif // MAPNIK_RASTER_COLORIZER_HPP
